import pandas as pd
import json
import os
from datetime import datetime


# json转为excel
def json_to_excel(input_json, output_excel):
    """
    将JSON文件转换为Excel文件
    参数:
        input_json: 输入的JSON文件路径
        output_excel: 输出的Excel文件路径
    """
    try:
        # 读取JSON文件
        with open(input_json, 'r', encoding='utf-8') as f:
            data = json.load(f)

        # 转换为DataFrame
        df = pd.DataFrame(data)

        # 写入Excel文件
        df.to_excel(output_excel, index=False, engine='openpyxl')

        print(f"成功将JSON转换为Excel，文件已保存到: {output_excel}")
        print(f"共转换 {len(df)} 条记录")

    except Exception as e:
        print(f"转换过程中发生错误: {str(e)}")


if __name__ == "__main__":
    # 输入输出文件路径
    input_file = "../json/response1.json"
    output_file = "../json/stock_data.xlsx"

    # 检查输入文件是否存在
    if not os.path.exists(input_file):
        print(f"错误: 输入文件 {input_file} 不存在")
    else:
        # 执行转换
        json_to_excel(input_file, output_file)
